package two_pointer.leetcode_15;

import dp.leetcode_198_easy.Rob;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class ThreeSum {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        if(nums == null || nums.length == 0){
            return res;
        }
        //排序
        Arrays.sort(nums);
        for (int i = 0; i < nums.length; i++){
            //nums[i] 已经 > 0，因为已经排序，所以不可能出现三数之和为 0 的情况，结束循环。
            if(nums[i] > 0){
                break;
            }
            //如果 nums[i] == nums[i - 1]，应跳过重复元素，因为该数所出现的情况已经寻找过了。
            if(i > 0 && nums[i] == nums[i - 1]){
                continue;
            }
            //双指针
            int left = i + 1;
            int right = nums.length - 1;
            while (left < right){
                int sum = nums[i] + nums[left] + nums[right];
                if(sum == 0){
                    res.add(Arrays.asList(nums[i], nums[left], nums[right]));
                    //去重
                    while (left < right && nums[left] == nums[left + 1]){
                        left++;
                    }
                    while (left < right && nums[right] == nums[right - 1]){
                        right--;
                    }
                    //如果有重复元素，去重后 left 和 right 处于最后一个重复元素的位置，继续缩小区间
                    left++;
                    right--;
                }else if(sum > 0){
                    right--;
                }else {
                    left++;
                }
            }
        }
        return res;
    }
}
